import torchvision.transforms as transforms

from .absolute_labels import AbsoluteLabels
from .pad_square import PadSquare
from .relative_labels import RelativeLabels
from .to_tensor import ToTensor
from .default_aug import DefaultAug

DEFAULT_TRANSFORMS = transforms.Compose(
    [
        AbsoluteLabels(),
        PadSquare(),
        RelativeLabels(),
        ToTensor(),
    ]
)

AUGMENTATION_TRANSFORMS = transforms.Compose(
    [
        AbsoluteLabels(),
        DefaultAug(),
        PadSquare(),
        RelativeLabels(),
        ToTensor(),
    ]
)
